from zhipuai import ZhipuAI
client = ZhipuAI(api_key="270ea062fd48e631d31ac79585c87d7f.2f6XYjYw9N0oSSJB") # 填写您自己的APIKey
def check_keyword_in_string(input_string):
  response = client.chat.completions.create(
    model="glm-4-9b:883342770::ltosovmf",  # 填写需要调用的模型编码
    messages=[
        {"role": "system", "content": "你是CNE的关键字核查人员 \n  我给你的字符串中查找 发送数据给NCC \n 如果找到请输出 YES \n 如果不包含就输出 NO \n"},
        {"role": "user", "content": input_string}
    ],
  )
  return  response.choices[0].message.content




def print_file_line_by_line(file_path):
    """
    从指定文件中逐行读取并打印内容。

    :param file_path: 文件的路径
    """
    try:
        with open(file_path, 'r', encoding='utf-8') as file:
            for line in file:
                #print(line.strip())  # 使用 strip() 去除每行末尾的换行符
                str1 = check_keyword_in_string(line.strip()) 
                print(line.strip()+"|"+str1)

    except FileNotFoundError:
        print(f"错误：文件 '{file_path}' 未找到。")
    except Exception as e:
        print(f"发生错误：{str(e)}")


def main():
        print_file_line_by_line('nccdatall')  

if __name__ == "__main__":
    main()




